eb9bf8484d228a7ea2914b79902161dfeaf995bf,deeplearning4j-core/src/main/java/org/deeplearning4j/parallelism/MagicQueue.java,MagicQueue,poll,#number#TimeUnit#,258
Before Change
if (numberOfBuckets > 1) {
int deviceId = Nd4j.getAffinityManager().getDeviceForCurrentThread();
return backingQueues.get(deviceId).poll(time, timeUnit);
} else return backingQueues.get(0).poll(time, timeUnit);
} else {
DataSet ds = backingQueues.get(interleavedCounter.getAndIncrement()).poll(time, timeUnit);
if (interleavedCounter.get() >= backingQueues.size())
After Change
return ds;
} else {
DataSet ds = backingQueues.get(0).poll(time, timeUnit);
if (ds != null)
cntGet.incrementAndGet();
return ds;
}
} else {
DataSet ds = backingQueues.get(interleavedCounter.getAndIncrement()).poll(time, timeUnit);